from selenium import webdriver
import time
from selenium.webdriver import ActionChains
from hashlib import md5
import requests


class Chaojiying_Client(object):

    def __init__(self, username, password, soft_id):
        self.username = username
        password = password.encode('utf8')
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files,
                          headers=self.headers)
        return r.json()

    def PostPic_base64(self, base64_str, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
            'file_base64': base64_str
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, headers=self.headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
        return r.json()


def yan_zheng_ma(driver):
    for i in range(1, 6):
        chaojiying = Chaojiying_Client('Future119', 'p0lqv59l', '971377')

        # 判断该界面有没有验证窗口
        red_btn = driver.find_elements("xpath", "//div[@class='dialog-footer']/button[@class='el-button el-button--red']")
        # 长度大于0，则表示有
        if len(red_btn) > 0:
            red_btn[0].click()
            time.sleep(3)
            code_element = driver.find_element("xpath", "//div[@class='geetest_holder geetest_silver']")
            code_element.screenshot("验证码.png")
            im = open('验证码.png', 'rb').read()
            xy_list = chaojiying.PostPic(im, 9004)["pic_str"]
            for xy in xy_list.split("|"):
                x, y = xy.split(",")
                x = int(x)
                y = int(y)
                ActionChains(driver).move_to_element_with_offset(code_element, x, y).click().perform()
                time.sleep(1)
            driver.find_element("xpath", "//div[@class='geetest_commit_tip']").click()
        time.sleep(10)
        #  重试的结束条件，最多重试5次
        if "重新验证" not in driver.page_source:
            break
    return
